write a function hackerrank solution